import numpy as np
from src.go import Go

if __name__ == '__main__':
    phase = [
        [0, 0, -1, 1, 1, 0, 1, 1, 1, -1, 0, -1, 0, -1, 1, 0, 1, 0, 0],
        [0, 0, -1, 1, 0, 1, 1, 1, -1, 0, -1, 0, -1, -1, 1, 1, 1, 0, 0],
        [0, 0, -1, 1, 1, 0, 0, 1, -1, -1, 0, -1, 0, -1, 1, 0, 1, 0, 0],
        [0, -1, 1, 0, 1, 1, 0, 1, -1, 0, -1, -1, -1, 0, -1, 1, 0, 1, 0],
        [0, -1, 1, 1, 0, 1, 1, -1, -1, -1, -1, 0, 0, -1, -1, 1, 0, 1, 0],
        [0, 0, -1, 1, 1, 1, 1, -1, 0, -1, 0, -1, 0, 0, -1, 1, 0, 0, 0],
        [0, 0, -1, -1, -1, -1, 1, -1, -1, 0, 0, 0, 0, 0, -1, 1, 0, 0, 0],
        [-1, -1, -1, 0, 0, 0, -1, 1, 1, -1, 0, -1, 0, 0, -1, 1, 0, 0, 0],
        [1, 1, -1, 0, 0, 0, -1, -1, 1, -1, 0, 0, 0, 0, -1, 1, 0, 0, 0],
        [1, 0, 1, -1, 0, -1, 1, 1, 1, 1, -1, -1, -1, 0, -1, 1, 1, 1, 1],
        [0, 1, 1, -1, 0, -1, 1, 0, 0, 0, 1, -1, 0, -1, -1, 1, 0, 0, 1],
        [1, 1, -1, -1, -1, -1, -1, 1, 0, 0, 1, -1, -1, 0, -1, 1, 0, 0, 0],
        [-1, -1, 0, -1, -1, 0, -1, 1, 0, 0, 1, -1, 0, -1, -1, -1, 1, 0, 0],
        [0, -1, 0, 0, 0, 0, -1, 1, 0, 1, 1, -1, -1, 0, -1, 1, 0, 0, 0],
        [0, -1, 0, 0, 0, -1, 1, 0, 0, 1, 0, 1, 1, -1, -1, 1, 0, 0, 0],
        [0, 0, -1, 0, -1, -1, 1, 1, 1, 1, 0, 1, 0, 1, 1, 0, 0, 0, 0],
        [0, -1, -1, 0, -1, 1, 0, 0, 0, 0, 1, 0, 0, 0, 0, 1, 1, 0, 0],
        [0, 0, -1, 0, -1, 1, 0, 1, 1, 0, 0, 1, 0, 1, 0, 1, 0, 0, 0],
        [0, 0, 0, -1, 1, 1, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0, 0, 0]
    ]

    go = Go()
    res = go.check_winner(phase)
    print("res ->", res)
